tensor([[[0.3100, 0.3100],
         [0.3100, 0.3100]],

        [[0.4100, 0.4100],
         [0.4100, 0.4100]],

        [[0.2800, 0.2800],
         [0.2800, 0.2800]]], grad_fn=<RoundBackward1>)